from flask import Flask, session, request, render_template, views, flash, redirect, url_for
# import flask
app = Flask(__name__)

# генерация ключа для работы с сессией
app.secret_key = "b8l\x0b~\xe3<y\xc5D\xb6v9\'D\xd0y\xeap\x80e\xcb&_\xc27"

@app.route('/', methods=['GET', 'POST'])
def login():
    if request.method == 'GET':
        return render_template('login.html')
    else:
        if 'logout' in request.form.values():
            session.pop('username', None)
            return redirect(url_for('login'))
        username = request.form['username']
        username = username.strip()     # убираем пробелы
        if username:                    # не пустая строка
            session['username'] = username
            return redirect('/welcome/')
        else:
            flash("Некорректно введено имя")
            return redirect(url_for('login'))
        
@app.route('/welcome/', methods=['GET', 'POST'])        
def welcome():
    if request.method == 'GET':
        if session:
            return render_template('welcome.html')
        else:
            flash("Для доступа к запрашиваемой странице необходимо ввести логин")
            return redirect(url_for('login'))
    else:
        pass

if __name__ == '__main__':
        app.debug = True                # режим отладки
        app.run()
